# 实现一个命令行版本的学生管理系统
import os
import sys

students = []

def save():
    '''
    用于存档
    :return:
    '''
    with open('record.txt', 'w', encoding='utf-8') as f:
        for s in students:
            f.write(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}\n")
        print(f'[存档成功] 共存储了{len(students)}条记录！')

def load():
    '''
    用户读档
    :return:
    '''
    # 如果存档文件不存在，则直接跳过读档流程
    if not os.path.exists('record.txt'):
        return

    # 独挡的时候要保证先把旧的数据给清理干净！
    global students
    students = []
    with open('record.txt', 'r', encoding='utf-8') as f:
        for line in f:
            line = line.strip()
            tokens = line.split('\t')
            if len(tokens) != 4:
                print(f'当前行格式存在问题！line={line}')
                continue
            student = {
                'studentId': tokens[0],
                'name': tokens[1],
                'gender': tokens[2],
                'className': tokens[3]
            }
            students.append(student)
    print(f'[读档成功] 共读取了{len(students)}条记录！')

def menu():
    print(' 1. 新增学生')
    print(' 2. 显示学生')
    print(' 3. 查找学生')
    print(' 4. 删除学生')
    print(' 0. 退出程序')
    choice = input('请输入您的选择：')
    return choice

def insert():
    print('[新增学生]开始!')
    studentId = input('请输入学生的学号：')
    name = input('请输入学生的姓名：')
    gender = input('请输入学生性别：')
    if gender not in ('男', '女'):
        print('性别输入的内容不符合要求，新增失败！')
        return
    className = input('请输入学生的班级：')

    # 使用一个字典把上述的信息给聚合起来
    student = {
        'studentId': studentId,
        'name': name,
        'gender': gender,
        'className':className
    }
    global students
    students.append(student)
    save()
    print('[新增学生]完毕!')

def show():
    # 遍历全局变量的这个列表，把每个学生的信息给打印出来
    print('[显示学生]开始!')
    for s in students:
        print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
    print(f'[显示学生]完毕！共显示了{len(students)}条数据！')

def find():
    # 根据学生姓名，来进行查找
    print('[查找学生]开始！')
    name = input('请输入要查找的同学姓名：')
    count = 0
    for s in students:
        if name == s['name']:
            print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
            count += 1
    print(f'[查找学生]结束！共找到了{count}个匹配的同学！')

def delete():
    print('[删除学生]开始！')
    studentId = input('请输入要删除的学生学号：')
    for s in students:
        if studentId == s['studentId']:
            print(f"删除{s['name']}同学的信息！")
            students.remove(s)
    save()
    print('[删除学生]结束！')


def main():
    '''
    入口函数
    :return:
    '''
    # 通过控制台和用户进行交互
    print('-----------------------------------------')
    print('            欢迎来到学生管理系统             ')
    print('-----------------------------------------')
    load()
    while True:
        # 通过 menu 函数来打印出菜单项
        choice = menu()
        if choice == '1':
            #新增学生
            insert()
        elif choice == '2':
            #显示学生
            show()
        elif choice == '3':
            #查找学生
            find()
        elif choice == '4':
            #删除学生
            delete()
        elif choice == '0':
            #退出程序
            print('goodbye!')
            sys.exit(0)
        else:
            print('您的输入有误！请重新输入！')

main()